#pragma once
#include "cpu/vision.h"

#ifdef WITH_CUDA
#include "cuda/vision.h"
#endif


at::Tensor pred_rand_gt(const at::Tensor& pred_gt_loss, const at::Tensor& mask, const float ratio) {
    if(pred_gt_loss.type().is_cuda() && mask.type().is_cuda()) {
#ifdef WITH_CUDA
    return rand_match_gt_cuda(pred_gt_loss, mask, ratio);
#else
    AT_ERROR("Not Compatible with CPU");
#endif
    }
    AT_ERROR("Not Implement on CPU");
}
